import zipfile
import os


# 将从notion获取的压缩包进行自动解压操作
def upzip_file(zip_file, target_dir):
    list = []
    # 打开需要解压的压缩包
    with zipfile.ZipFile(zip_file, "r") as zfile:
        # 循环压缩包里的每个文件
        for file in zfile.namelist():
            # 判断只解压需要的csv文件
            if file[-4:] == ".csv":
                # 将解压出来的csv文件名存入到列表，方便后续的操作
                list.append("/" + file)
                # 对选定的csv文件进行解压操作
                zfile.extract(file, target_dir)
    return list


def csv_file_name(file):
    file_path = file
    return os.listdir(file_path)


if __name__ == "__main__":
    print(csv_file_name("/data/1003\\"))
